#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
PaddleOCR安装脚本
"""

import subprocess
import sys

def install_paddleocr():
    """安装PaddleOCR"""
    print("正在安装PaddleOCR...")
    print("这可能需要几分钟时间，请耐心等待...")
    print()
    
    try:
        # 使用国内镜像源安装
        subprocess.check_call([
            sys.executable, "-m", "pip", "install", 
            "paddleocr", "-i", "https://pypi.tuna.tsinghua.edu.cn/simple/"
        ])
        print("✅ PaddleOCR安装成功！")
        return True
    except subprocess.CalledProcessError as e:
        print(f"❌ PaddleOCR安装失败: {e}")
        return False

def main():
    print("=" * 50)
    print("    PaddleOCR 安装工具")
    print("=" * 50)
    print()
    
    if install_paddleocr():
        print("\n安装完成！现在可以运行物体识别程序了。")
    else:
        print("\n安装失败，但程序仍可以运行（OCR功能不可用）。")
    
    input("\n按回车键退出...")

if __name__ == "__main__":
    main() 